package io.renren.modules.app.controller;

import io.renren.common.utils.JsonResult;
import io.renren.common.utils.R;
import io.renren.modules.app.annotation.Login;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.multipart.MultipartFile;

import javax.servlet.http.HttpServletRequest;
import java.io.File;
import java.io.IOException;
import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.UUID;

@Slf4j
@RestController
@RequestMapping("app/upload")
public class UploadFilesController {

    /**
     * 图片保存路径，自动从yml文件中获取数据
     * 示例： E:/images/
     */
    @Value("${file-save-path}")
    private String fileSavePath;

    /**
     * 时间格式化
     */
    private SimpleDateFormat simpleDateFormat = new SimpleDateFormat("yyyy/MM/dd/");

    /**
     * 上传头像
     *
     * @param file    文件名称
     * @param request servlet 请求
     * @return 图片信息
     */
    @Login
    @PostMapping("/uploadAvatar")
    public R uploadAvatar(MultipartFile file, HttpServletRequest request) {
        String directory = simpleDateFormat.format(new Date());
        //文件保存目录  E:/images/日期 如果目录不存在，则创建
        File dir = new File(fileSavePath + directory);
        if (!dir.exists()) {
            dir.mkdirs();
        }
        //后缀
        String suffix = file.getOriginalFilename().substring(file.getOriginalFilename().lastIndexOf("."));
        String newFileName = UUID.randomUUID().toString().replaceAll("-", "") + suffix;
        //创建这个新文件
        File newFile = new File(fileSavePath + directory + newFileName);
        try {
            file.transferTo(newFile);
            //协议 :// ip地址 ：端口号 / 文件目录(/images/2020/03/15/xxx.jpg)
            String url = request.getScheme() + "://" + request.getServerName() + ":" + request.getServerPort() + "/images/" + directory + newFileName;
            return R.ok().put("imageUrl", url);
        } catch (IOException e) {
            return R.error("IO异常");
        }
    }

    /**
     * 上传充值凭证图
     *
     * @param file    文件名称
     * @param request servlet 请求
     * @return 图片信息
     */
    @Login
    @PostMapping("/uploadCertificateImg")
    public R uploadCertificateImg(MultipartFile file, HttpServletRequest request) {
        String directory = simpleDateFormat.format(new Date());
        File dir = new File(fileSavePath + directory);
        if (!dir.exists()) {
            dir.mkdirs();
        }
        //后缀
        String suffix = file.getOriginalFilename().substring(file.getOriginalFilename().lastIndexOf("."));
        String newFileName = UUID.randomUUID().toString().replaceAll("-", "") + suffix;
        //创建这个新文件
        File newFile = new File(fileSavePath  + directory + newFileName);
        try {
            file.transferTo(newFile);
            String url = request.getScheme() + "://" + request.getServerName() + ":" + request.getServerPort() + "/images/"  + directory + newFileName;
            System.out.println("url-" + url);
            return R.ok().put("imageUrl", url);
        } catch (IOException e) {
            return R.error("IO异常");
        }
    }

}
